import pandas as pd
import matplotlib.pyplot as plt

# 读取CSV文件
data = pd.read_csv("cleaned_data.csv")

# 统计不同season对应的count总和
season_counts = data.groupby('season')['count'].sum()

# 设置中文字体
plt.rcParams['font.sans-serif'] = ['SimHei']  # 设置中文显示
plt.rcParams['axes.unicode_minus'] = False    # 解决保存图像是负号'-'显示为方块的问题

# 绘制条形图并设置颜色为紫色
bars = plt.bar(season_counts.index, season_counts.values, color='purple')
plt.xlabel('季节')
plt.ylabel('租用数')
plt.title('不同季节共享单车租用数')

# 在每个条形上方显示具体数值
for bar in bars:
    yval = bar.get_height()
    plt.text(bar.get_x() + bar.get_width()/3.0, yval, int(yval), va='bottom')  # va='bottom'使得数字在柱子顶部显示

plt.xticks(season_counts.index, ['春季', '夏季', '秋季', '冬季'])  # 将数字标签改为对应的季节
plt.show()